package designPattern.structuralPattern.composite;

/**
 * @author wei.zhao
 * @description:
 * @date 2021年12月16日 17:23
 */
public class Test {

    public static void main(String[] args) {
        //总文件夹
        Folder folder = new Folder("总文件夹");
        //向总文件夹中放入三个文件：a.txt、b.jpg、c文件夹
        TextFile aText = new TextFile("a.txt");
        ImageFile bImage = new ImageFile("b.jpg");
        Folder cFolder = new Folder("c文件夹");
        folder.add(aText);
        folder.add(bImage);
        folder.add(cFolder);
        //向c文件夹中添加文件：c1.txt、c1.jpg
        TextFile cText = new TextFile("c1.txt");
        ImageFile cImage = new ImageFile("c1.jpg");
        cFolder.add(cText);
        cFolder.add(cImage);
        //打印所有文件
        folder.display();
    }

}
